from Broswer import *
import unittest

class TestBaidu(unittest.TestCase):

    def test_search_key_selenium(self):
        serach_key = "selenium"
        baidu_search(serach_key)
        self.assertEqual(Browser.driver.title, serach_key + "_百度搜索")

    def test_search_key_unttest(self):
        serach_key = "unttest"
        baidu_search(serach_key)
        self.assertEqual(Browser.driver.title, serach_key + "_百度搜索")